package com.rslai.base.tool.servertest.database.postgresql;

import java.sql.Connection;

import org.dbunit.DatabaseUnitException;
import org.dbunit.database.DatabaseConfig;
import org.dbunit.database.DatabaseConnection;

public class PostgresConnection extends DatabaseConnection {
    public PostgresConnection(Connection connection, String schema) throws DatabaseUnitException {
        super(connection, schema);
        PostgresqlDataTypeFactory dataTypeFactory = new PostgresqlDataTypeFactory();
        getConfig().setProperty("http://www.dbunit.org/properties/datatypeFactory", dataTypeFactory);
        getConfig().setProperty("http://www.dbunit.org/properties/escapePattern", "\"");
    }
}
